1 /* 2 * Copyright (C) 2007 The Guava Authors 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package com.google.common.collect; 18 19 import static com.google.common.base.Preconditions.checkArgument; 20 import static com.google.common.base.Preconditions.checkNotNull; 21 22 import com.google.common.annotations.GwtCompatible; 23 import com.google.common.base.Predicate; 24 import com.google.common.base.Predicates; 25 import com.google.common.collect.Collections2.FilteredCollection; 26 27 import java.util.AbstractSet; 28 import java.util.Arrays; 29 import java.util.Collection; 30 import java.util.Collections; 31 import java.util.Comparator; 32 import java.util.EnumSet; 33 import java.util.HashSet; 34 import java.util.Iterator; 35 import java.util.LinkedHashSet; 36 import java.util.List; 37 import java.util.Map; 38 import java.util.NoSuchElementException; 39 import java.util.Set; 40 import java.util.SortedSet; 41 import java.util.TreeSet; 42 import java.util.concurrent.ConcurrentHashMap; 43 44 import javax.annotation.Nullable; 45 46 /** 47 * Static utility methods pertaining to {@link Set} instances. Also see this 48 * class's counterparts {@link Lists}, {@link Maps} and {@link Queues}. 49 * 50 * <p>See the Guava User Guide article on <a href= 51 * "http://code.google.com/p/guava-libraries/wiki/CollectionUtilitiesExplained#Sets"> 52 * {@code Sets}</a>. 53 * 54 * @author Kevin Bourrillion 55 * @author Jared Levy 56 * @author Chris Povirk 57 * @since 2.0 (imported from Google Collections Library) 58 */ 59 @GwtCompatible(emulated = true) 60 public final class Sets { 61 private Sets() {} 62 63 /** 64 * {@link AbstractSet} substitute without the potentially-quadratic 65 * {@code removeAll} implementation. 66 */ 67 abstract static class ImprovedAbstractSet<E> extends AbstractSet<E> { 68 @Override 69 public boolean removeAll(Collection<?> c) { 70 return removeAllImpl(this, c); 71 } 72 73 @Override 74 public boolean retainAll(Collection<?> c) { 75 return super.retainAll(checkNotNull(c)); // GWT compatibility 76 } 77 } 78 79 /** 80 * Returns an immutable set instance containing the given enum elements. 81 * Internally, the returned set will be backed by an {@link EnumSet}. 82 * 83 * <p>The iteration order of the returned set follows the enum's iteration 84 * order, not the order in which the elements are provided to the method. 85 * 86 * @param anElement one of the elements the set should contain 87 * @param otherElements the rest of the elements the set should contain 88 * @return an immutable set containing those elements, minus duplicates 89 */ 90 // http://code.google.com/p/google-web-toolkit/issues/detail?id=3028 91 @GwtCompatible(serializable = true) 92 public static <E extends Enum<E>> ImmutableSet<E> immutableEnumSet( 93 E anElement, E... otherElements) { 94 return ImmutableEnumSet.asImmutable(EnumSet.of(anElement, otherElements)); 95 } 96 97 /** 98 * Returns an immutable set instance containing the given enum elements. 99 * Internally, the returned set will be backed by an {@link EnumSet}. 100 * 101 * <p>The iteration order of the returned set follows the enum's iteration 102 * order, not the order in which the elements appear in the given collection. 103 * 104 * @param elements the elements, all of the same {@code enum} type, that the 105 * set should contain 106 * @return an immutable set containing those elements, minus duplicates 107 */ 108 // http://code.google.com/p/google-web-toolkit/issues/detail?id=3028 109 @GwtCompatible(serializable = true) 110 public static <E extends Enum<E>> ImmutableSet<E> immutableEnumSet( 111 Iterable<E> elements) { 112 if (elements instanceof ImmutableEnumSet) { 113 return (ImmutableEnumSet<E>) elements; 114 } else if (elements instanceof Collection) { 115 Collection<E> collection = (Collection<E>) elements; 116 if (collection.isEmpty()) { 117 return ImmutableSet.of(); 118 } else { 119 return ImmutableEnumSet.asImmutable(EnumSet.copyOf(collection)); 120 } 121 } else { 122 Iterator<E> itr = elements.iterator(); 123 if (itr.hasNext()) { 124 EnumSet<E> enumSet = EnumSet.of(itr.next()); 125 Iterators.addAll(enumSet, itr); 126 return ImmutableEnumSet.asImmutable(enumSet); 127 } else { 128 return ImmutableSet.of(); 129 } 130 } 131 } 132 133 /** 134 * Returns a new {@code EnumSet} instance containing the given elements. 135 * Unlike {@link EnumSet#copyOf(Collection)}, this method does not produce an 136 * exception on an empty collection, and it may be called on any iterable, not 137 * just a {@code Collection}. 138 */ 139 public static <E extends Enum<E>> EnumSet<E> newEnumSet(Iterable<E> iterable, 140 Class<E> elementType) { 141 EnumSet<E> set = EnumSet.noneOf(elementType); 142 Iterables.addAll(set, iterable); 143 return set; 144 } 145 146 // HashSet 147 148 /** 149 * Creates a <i>mutable</i>, empty {@code HashSet} instance. 150 * 151 * <p><b>Note:</b> if mutability is not required, use {@link 152 * ImmutableSet#of()} instead. 153 * 154 * <p><b>Note:</b> if {@code E} is an {@link Enum} type, use {@link 155 * EnumSet#noneOf} instead. 156 * 157 * @return a new, empty {@code HashSet} 158 */ 159 public static <E> HashSet<E> newHashSet() { 160 return new HashSet<E>(); 161 } 162 163 /** 164 * Creates a <i>mutable</i> {@code HashSet} instance containing the given 165 * elements in unspecified order. 166 * 167 * <p><b>Note:</b> if mutability is not required and the elements are 168 * non-null, use an overload of {@link ImmutableSet#of()} (for varargs) or 169 * {@link ImmutableSet#copyOf(Object[])} (for an array) instead. 170 * 171 * <p><b>Note:</b> if {@code E} is an {@link Enum} type, use {@link 172 * EnumSet#of(Enum, Enum[])} instead. 173 * 174 * @param elements the elements that the set should contain 175 * @return a new {@code HashSet} containing those elements (minus duplicates) 176 */ 177 public static <E> HashSet<E> newHashSet(E... elements) { 178 HashSet<E> set = newHashSetWithExpectedSize(elements.length); 179 Collections.addAll(set, elements); 180 return set; 181 } 182 183 /** 184 * Creates a {@code HashSet} instance, with a high enough "initial capacity" 185 * that it <i>should</i> hold {@code expectedSize} elements without growth. 186 * This behavior cannot be broadly guaranteed, but it is observed to be true 187 * for OpenJDK 1.6. It also can't be guaranteed that the method isn't 188 * inadvertently <i>oversizing</i> the returned set. 189 * 190 * @param expectedSize the number of elements you expect to add to the 191 * returned set 192 * @return a new, empty {@code HashSet} with enough capacity to hold {@code 193 * expectedSize} elements without resizing 194 * @throws IllegalArgumentException if {@code expectedSize} is negative 195 */ 196 public static <E> HashSet<E> newHashSetWithExpectedSize(int expectedSize) { 197 return new HashSet<E>(Maps.capacity(expectedSize)); 198 } 199 200 /** 201 * Creates a <i>mutable</i> {@code HashSet} instance containing the given 202 * elements in unspecified order. 203 * 204 * <p><b>Note:</b> if mutability is not required and the elements are 205 * non-null, use {@link ImmutableSet#copyOf(Iterable)} instead. 206 * 207 * <p><b>Note:</b> if {@code E} is an {@link Enum} type, use 208 * {@link #newEnumSet(Iterable, Class)} instead. 209 * 210 * @param elements the elements that the set should contain 211 * @return a new {@code HashSet} containing those elements (minus duplicates) 212 */ 213 public static <E> HashSet<E> newHashSet(Iterable<? extends E> elements) { 214 return (elements instanceof Collection) 215 ? new HashSet<E>(Collections2.cast(elements)) 216 : newHashSet(elements.iterator()); 217 } 218 219 /** 220 * Creates a <i>mutable</i> {@code HashSet} instance containing the given 221 * elements in unspecified order. 222 * 223 * <p><b>Note:</b> if mutability is not required and the elements are 224 * non-null, use {@link ImmutableSet#copyOf(Iterable)} instead. 225 * 226 * <p><b>Note:</b> if {@code E} is an {@link Enum} type, you should create an 227 * {@link EnumSet} instead. 228 * 229 * @param elements the elements that the set should contain 230 * @return a new {@code HashSet} containing those elements (minus duplicates) 231 */ 232 public static <E> HashSet<E> newHashSet(Iterator<? extends E> elements) { 233 HashSet<E> set = newHashSet(); 234 Iterators.addAll(set, elements); 235 return set; 236 } 237 238 /** 239 * Creates a thread-safe set backed by a hash map. The set is backed by a 240 * {@link ConcurrentHashMap} instance, and thus carries the same concurrency 241 * guarantees. 242 * 243 * <p>Unlike {@code HashSet}, this class does NOT allow {@code null} to be 244 * used as an element. The set is serializable. 245 * 246 * @return a new, empty thread-safe {@code Set} 247 * @since 15.0 248 */ 249 public static <E> Set<E> newConcurrentHashSet() { 250 return newSetFromMap(new ConcurrentHashMap<E, Boolean>()); 251 } 252 253 /** 254 * Creates a thread-safe set backed by a hash map and containing the given 255 * elements. The set is backed by a {@link ConcurrentHashMap} instance, and 256 * thus carries the same concurrency guarantees. 257 * 258 * <p>Unlike {@code HashSet}, this class does NOT allow {@code null} to be 259 * used as an element. The set is serializable. 260 * 261 * @param elements the elements that the set should contain 262 * @return a new thread-safe set containing those elements (minus duplicates) 263 * @throws NullPointerException if {@code elements} or any of its contents is 264 * null 265 * @since 15.0 266 */ 267 public static <E> Set<E> newConcurrentHashSet( 268 Iterable<? extends E> elements) { 269 Set<E> set = newConcurrentHashSet(); 270 Iterables.addAll(set, elements); 271 return set; 272 } 273 274 // LinkedHashSet 275 276 /** 277 * Creates a <i>mutable</i>, empty {@code LinkedHashSet} instance. 278 * 279 * <p><b>Note:</b> if mutability is not required, use {@link 280 * ImmutableSet#of()} instead. 281 * 282 * @return a new, empty {@code LinkedHashSet} 283 */ 284 public static <E> LinkedHashSet<E> newLinkedHashSet() { 285 return new LinkedHashSet<E>(); 286 } 287 288 /** 289 * Creates a {@code LinkedHashSet} instance, with a high enough "initial 290 * capacity" that it <i>should</i> hold {@code expectedSize} elements without 291 * growth. This behavior cannot be broadly guaranteed, but it is observed to 292 * be true for OpenJDK 1.6. It also can't be guaranteed that the method isn't 293 * inadvertently <i>oversizing</i> the returned set. 294 * 295 * @param expectedSize the number of elements you expect to add to the 296 * returned set 297 * @return a new, empty {@code LinkedHashSet} with enough capacity to hold 298 * {@code expectedSize} elements without resizing 299 * @throws IllegalArgumentException if {@code expectedSize} is negative 300 * @since 11.0 301 */ 302 public static <E> LinkedHashSet<E> newLinkedHashSetWithExpectedSize( 303 int expectedSize) { 304 return new LinkedHashSet<E>(Maps.capacity(expectedSize)); 305 } 306 307 /** 308 * Creates a <i>mutable</i> {@code LinkedHashSet} instance containing the 309 * given elements in order. 310 * 311 * <p><b>Note:</b> if mutability is not required and the elements are 312 * non-null, use {@link ImmutableSet#copyOf(Iterable)} instead. 313 * 314 * @param elements the elements that the set should contain, in order 315 * @return a new {@code LinkedHashSet} containing those elements (minus 316 * duplicates) 317 */ 318 public static <E> LinkedHashSet<E> newLinkedHashSet( 319 Iterable<? extends E> elements) { 320 if (elements instanceof Collection) { 321 return new LinkedHashSet<E>(Collections2.cast(elements)); 322 } 323 LinkedHashSet<E> set = newLinkedHashSet(); 324 Iterables.addAll(set, elements); 325 return set; 326 } 327 328 // TreeSet 329 330 /** 331 * Creates a <i>mutable</i>, empty {@code TreeSet} instance sorted by the 332 * natural sort ordering of its elements. 333 * 334 * <p><b>Note:</b> if mutability is not required, use {@link 335 * ImmutableSortedSet#of()} instead. 336 * 337 * @return a new, empty {@code TreeSet} 338 */ 339 public static <E extends Comparable> TreeSet<E> newTreeSet() { 340 return new TreeSet<E>(); 341 } 342 343 /** 344 * Creates a <i>mutable</i> {@code TreeSet} instance containing the given 345 * elements sorted by their natural ordering. 346 * 347 * <p><b>Note:</b> if mutability is not required, use {@link 348 * ImmutableSortedSet#copyOf(Iterable)} instead. 349 * 350 * <p><b>Note:</b> If {@code elements} is a {@code SortedSet} with an explicit 351 * comparator, this method has different behavior than 352 * {@link TreeSet#TreeSet(SortedSet)}, which returns a {@code TreeSet} with 353 * that comparator. 354 * 355 * @param elements the elements that the set should contain 356 * @return a new {@code TreeSet} containing those elements (minus duplicates) 357 */ 358 public static <E extends Comparable> TreeSet<E> newTreeSet( 359 Iterable<? extends E> elements) { 360 TreeSet<E> set = newTreeSet(); 361 Iterables.addAll(set, elements); 362 return set; 363 } 364 365 /** 366 * Creates a <i>mutable</i>, empty {@code TreeSet} instance with the given 367 * comparator. 368 * 369 * <p><b>Note:</b> if mutability is not required, use {@code 370 * ImmutableSortedSet.orderedBy(comparator).build()} instead. 371 * 372 * @param comparator the comparator to use to sort the set 373 * @return a new, empty {@code TreeSet} 374 * @throws NullPointerException if {@code comparator} is null 375 */ 376 public static <E> TreeSet<E> newTreeSet(Comparator<? super E> comparator) { 377 return new TreeSet<E>(checkNotNull(comparator)); 378 } 379 380 /** 381 * Creates an empty {@code Set} that uses identity to determine equality. It 382 * compares object references, instead of calling {@code equals}, to 383 * determine whether a provided object matches an element in the set. For 384 * example, {@code contains} returns {@code false} when passed an object that 385 * equals a set member, but isn't the same instance. This behavior is similar 386 * to the way {@code IdentityHashMap} handles key lookups. 387 * 388 * @since 8.0 389 */ 390 public static <E> Set<E> newIdentityHashSet() { 391 return Sets.newSetFromMap(Maps.<E, Boolean>newIdentityHashMap()); 392 } 393 394 /** 395 * Creates an {@code EnumSet} consisting of all enum values that are not in 396 * the specified collection. If the collection is an {@link EnumSet}, this 397 * method has the same behavior as {@link EnumSet#complementOf}. Otherwise, 398 * the specified collection must contain at least one element, in order to 399 * determine the element type. If the collection could be empty, use 400 * {@link #complementOf(Collection, Class)} instead of this method. 401 * 402 * @param collection the collection whose complement should be stored in the 403 * enum set 404 * @return a new, modifiable {@code EnumSet} containing all values of the enum 405 * that aren't present in the given collection 406 * @throws IllegalArgumentException if {@code collection} is not an 407 * {@code EnumSet} instance and contains no elements 408 */ 409 public static <E extends Enum<E>> EnumSet<E> complementOf( 410 Collection<E> collection) { 411 if (collection instanceof EnumSet) { 412 return EnumSet.complementOf((EnumSet<E>) collection); 413 } 414 checkArgument(!collection.isEmpty(), 415 "collection is empty; use the other version of this method"); 416 Class<E> type = collection.iterator().next().getDeclaringClass(); 417 return makeComplementByHand(collection, type); 418 } 419 420 /** 421 * Creates an {@code EnumSet} consisting of all enum values that are not in 422 * the specified collection. This is equivalent to 423 * {@link EnumSet#complementOf}, but can act on any input collection, as long 424 * as the elements are of enum type. 425 * 426 * @param collection the collection whose complement should be stored in the 427 * {@code EnumSet} 428 * @param type the type of the elements in the set 429 * @return a new, modifiable {@code EnumSet} initially containing all the 430 * values of the enum not present in the given collection 431 */ 432 public static <E extends Enum<E>> EnumSet<E> complementOf( 433 Collection<E> collection, Class<E> type) { 434 checkNotNull(collection); 435 return (collection instanceof EnumSet) 436 ? EnumSet.complementOf((EnumSet<E>) collection) 437 : makeComplementByHand(collection, type); 438 } 439 440 private static <E extends Enum<E>> EnumSet<E> makeComplementByHand( 441 Collection<E> collection, Class<E> type) { 442 EnumSet<E> result = EnumSet.allOf(type); 443 result.removeAll(collection); 444 return result; 445 } 446 447 /** 448 * Returns a set backed by the specified map. The resulting set displays 449 * the same ordering, concurrency, and performance characteristics as the 450 * backing map. In essence, this factory method provides a {@link Set} 451 * implementation corresponding to any {@link Map} implementation. There is no 452 * need to use this method on a {@link Map} implementation that already has a 453 * corresponding {@link Set} implementation (such as {@link java.util.HashMap} 454 * or {@link java.util.TreeMap}). 455 * 456 * <p>Each method invocation on the set returned by this method results in 457 * exactly one method invocation on the backing map or its {@code keySet} 458 * view, with one exception. The {@code addAll} method is implemented as a 459 * sequence of {@code put} invocations on the backing map. 460 * 461 * <p>The specified map must be empty at the time this method is invoked, 462 * and should not be accessed directly after this method returns. These 463 * conditions are ensured if the map is created empty, passed directly 464 * to this method, and no reference to the map is retained, as illustrated 465 * in the following code fragment: <pre> {@code 466 * 467 * Set<Object> identityHashSet = Sets.newSetFromMap( 468 * new IdentityHashMap<Object, Boolean>());}</pre> 469 * 470 * <p>This method has the same behavior as the JDK 6 method 471 * {@code Collections.newSetFromMap()}. The returned set is serializable if 472 * the backing map is. 473 * 474 * @param map the backing map 475 * @return the set backed by the map 476 * @throws IllegalArgumentException if {@code map} is not empty 477 */ 478 public static <E> Set<E> newSetFromMap(Map<E, Boolean> map) { 479 return Platform.newSetFromMap(map); 480 } 481 482 /** 483 * An unmodifiable view of a set which may be backed by other sets; this view 484 * will change as the backing sets do. Contains methods to copy the data into 485 * a new set which will then remain stable. There is usually no reason to 486 * retain a reference of type {@code SetView}; typically, you either use it 487 * as a plain {@link Set}, or immediately invoke {@link #immutableCopy} or 488 * {@link #copyInto} and forget the {@code SetView} itself. 489 * 490 * @since 2.0 (imported from Google Collections Library) 491 */ 492 public abstract static class SetView<E> extends AbstractSet<E> { 493 private SetView() {} // no subclasses but our own 494 495 /** 496 * Returns an immutable copy of the current contents of this set view. 497 * Does not support null elements. 498 * 499 * <p><b>Warning:</b> this may have unexpected results if a backing set of 500 * this view uses a nonstandard notion of equivalence, for example if it is 501 * a {@link TreeSet} using a comparator that is inconsistent with {@link 502 * Object#equals(Object)}. 503 */ 504 public ImmutableSet<E> immutableCopy() { 505 return ImmutableSet.copyOf(this); 506 } 507 508 /** 509 * Copies the current contents of this set view into an existing set. This 510 * method has equivalent behavior to {@code set.addAll(this)}, assuming that 511 * all the sets involved are based on the same notion of equivalence. 512 * 513 * @return a reference to {@code set}, for convenience 514 */ 515 // Note: S should logically extend Set<? super E> but can't due to either 516 // some javac bug or some weirdness in the spec, not sure which. 517 public <S extends Set<E>> S copyInto(S set) { 518 set.addAll(this); 519 return set; 520 } 521 } 522 523 /** 524 * Returns an unmodifiable <b>view</b> of the union of two sets. The returned 525 * set contains all elements that are contained in either backing set. 526 * Iterating over the returned set iterates first over all the elements of 527 * {@code set1}, then over each element of {@code set2}, in order, that is not 528 * contained in {@code set1}. 529 * 530 * <p>Results are undefined if {@code set1} and {@code set2} are sets based on 531 * different equivalence relations (as {@link HashSet}, {@link TreeSet}, and 532 * the {@link Map#keySet} of an {@code IdentityHashMap} all are). 533 * 534 * <p><b>Note:</b> The returned view performs better when {@code set1} is the 535 * smaller of the two sets. If you have reason to believe one of your sets 536 * will generally be smaller than the other, pass it first. 537 * 538 * <p>Further, note that the current implementation is not suitable for nested 539 * {@code union} views, i.e. the following should be avoided when in a loop: 540 * {@code union = Sets.union(union, anotherSet);}, since iterating over the resulting 541 * set has a cubic complexity to the depth of the nesting. 542 */ 543 public static <E> SetView<E> union( 544 final Set<? extends E> set1, final Set<? extends E> set2) { 545 checkNotNull(set1, "set1"); 546 checkNotNull(set2, "set2"); 547 548 final Set<? extends E> set2minus1 = difference(set2, set1); 549 550 return new SetView<E>() { 551 @Override public int size() { 552 return set1.size() + set2minus1.size(); 553 } 554 @Override public boolean isEmpty() { 555 return set1.isEmpty() && set2.isEmpty(); 556 } 557 @Override public Iterator<E> iterator() { 558 return Iterators.unmodifiableIterator( 559 Iterators.concat(set1.iterator(), set2minus1.iterator())); 560 } 561 @Override public boolean contains(Object object) { 562 return set1.contains(object) || set2.contains(object); 563 } 564 @Override public <S extends Set<E>> S copyInto(S set) { 565 set.addAll(set1); 566 set.addAll(set2); 567 return set; 568 } 569 @Override public ImmutableSet<E> immutableCopy() { 570 return new ImmutableSet.Builder<E>() 571 .addAll(set1).addAll(set2).build(); 572 } 573 }; 574 } 575 576 /** 577 * Returns an unmodifiable <b>view</b> of the intersection of two sets. The 578 * returned set contains all elements that are contained by both backing sets. 579 * The iteration order of the returned set matches that of {@code set1}. 580 * 581 * <p>Results are undefined if {@code set1} and {@code set2} are sets based 582 * on different equivalence relations (as {@code HashSet}, {@code TreeSet}, 583 * and the keySet of an {@code IdentityHashMap} all are). 584 * 585 * <p><b>Note:</b> The returned view performs slightly better when {@code 586 * set1} is the smaller of the two sets. If you have reason to believe one of 587 * your sets will generally be smaller than the other, pass it first. 588 * Unfortunately, since this method sets the generic type of the returned set 589 * based on the type of the first set passed, this could in rare cases force 590 * you to make a cast, for example: <pre> {@code 591 * 592 * Set<Object> aFewBadObjects = ... 593 * Set<String> manyBadStrings = ... 594 * 595 * // impossible for a non-String to be in the intersection 596 * SuppressWarnings("unchecked") 597 * Set<String> badStrings = (Set) Sets.intersection( 598 * aFewBadObjects, manyBadStrings);}</pre> 599 * 600 * <p>This is unfortunate, but should come up only very rarely. 601 */ 602 public static <E> SetView<E> intersection( 603 final Set<E> set1, final Set<?> set2) { 604 checkNotNull(set1, "set1"); 605 checkNotNull(set2, "set2"); 606 607 final Predicate<Object> inSet2 = Predicates.in(set2); 608 return new SetView<E>() { 609 @Override public Iterator<E> iterator() { 610 return Iterators.filter(set1.iterator(), inSet2); 611 } 612 @Override public int size() { 613 return Iterators.size(iterator()); 614 } 615 @Override public boolean isEmpty() { 616 return !iterator().hasNext(); 617 } 618 @Override public boolean contains(Object object) { 619 return set1.contains(object) && set2.contains(object); 620 } 621 @Override public boolean containsAll(Collection<?> collection) { 622 return set1.containsAll(collection) 623 && set2.containsAll(collection); 624 } 625 }; 626 } 627 628 /** 629 * Returns an unmodifiable <b>view</b> of the difference of two sets. The 630 * returned set contains all elements that are contained by {@code set1} and 631 * not contained by {@code set2}. {@code set2} may also contain elements not 632 * present in {@code set1}; these are simply ignored. The iteration order of 633 * the returned set matches that of {@code set1}. 634 * 635 * <p>Results are undefined if {@code set1} and {@code set2} are sets based 636 * on different equivalence relations (as {@code HashSet}, {@code TreeSet}, 637 * and the keySet of an {@code IdentityHashMap} all are). 638 */ 639 public static <E> SetView<E> difference( 640 final Set<E> set1, final Set<?> set2) { 641 checkNotNull(set1, "set1"); 642 checkNotNull(set2, "set2"); 643 644 final Predicate<Object> notInSet2 = Predicates.not(Predicates.in(set2)); 645 return new SetView<E>() { 646 @Override public Iterator<E> iterator() { 647 return Iterators.filter(set1.iterator(), notInSet2); 648 } 649 @Override public int size() { 650 return Iterators.size(iterator()); 651 } 652 @Override public boolean isEmpty() { 653 return set2.containsAll(set1); 654 } 655 @Override public boolean contains(Object element) { 656 return set1.contains(element) && !set2.contains(element); 657 } 658 }; 659 } 660 661 /** 662 * Returns an unmodifiable <b>view</b> of the symmetric difference of two 663 * sets. The returned set contains all elements that are contained in either 664 * {@code set1} or {@code set2} but not in both. The iteration order of the 665 * returned set is undefined. 666 * 667 * <p>Results are undefined if {@code set1} and {@code set2} are sets based 668 * on different equivalence relations (as {@code HashSet}, {@code TreeSet}, 669 * and the keySet of an {@code IdentityHashMap} all are). 670 * 671 * @since 3.0 672 */ 673 public static <E> SetView<E> symmetricDifference( 674 Set<? extends E> set1, Set<? extends E> set2) { 675 checkNotNull(set1, "set1"); 676 checkNotNull(set2, "set2"); 677 678 // TODO(kevinb): Replace this with a more efficient implementation 679 return difference(union(set1, set2), intersection(set1, set2)); 680 } 681 682 /** 683 * Returns the elements of {@code unfiltered} that satisfy a predicate. The 684 * returned set is a live view of {@code unfiltered}; changes to one affect 685 * the other. 686 * 687 * <p>The resulting set's iterator does not support {@code remove()}, but all 688 * other set methods are supported. When given an element that doesn't satisfy 689 * the predicate, the set's {@code add()} and {@code addAll()} methods throw 690 * an {@link IllegalArgumentException}. When methods such as {@code 691 * removeAll()} and {@code clear()} are called on the filtered set, only 692 * elements that satisfy the filter will be removed from the underlying set. 693 * 694 * <p>The returned set isn't threadsafe or serializable, even if 695 * {@code unfiltered} is. 696 * 697 * <p>Many of the filtered set's methods, such as {@code size()}, iterate 698 * across every element in the underlying set and determine which elements 699 * satisfy the filter. When a live view is <i>not</i> needed, it may be faster 700 * to copy {@code Iterables.filter(unfiltered, predicate)} and use the copy. 701 * 702 * <p><b>Warning:</b> {@code predicate} must be <i>consistent with equals</i>, 703 * as documented at {@link Predicate#apply}. Do not provide a predicate such 704 * as {@code Predicates.instanceOf(ArrayList.class)}, which is inconsistent 705 * with equals. (See {@link Iterables#filter(Iterable, Class)} for related 706 * functionality.) 707 */ 708 // TODO(kevinb): how to omit that last sentence when building GWT javadoc? 709 public static <E> Set<E> filter( 710 Set<E> unfiltered, Predicate<? super E> predicate) { 711 if (unfiltered instanceof SortedSet) { 712 return filter((SortedSet<E>) unfiltered, predicate); 713 } 714 if (unfiltered instanceof FilteredSet) { 715 // Support clear(), removeAll(), and retainAll() when filtering a filtered 716 // collection. 717 FilteredSet<E> filtered = (FilteredSet<E>) unfiltered; 718 Predicate<E> combinedPredicate 719 = Predicates.<E>and(filtered.predicate, predicate); 720 return new FilteredSet<E>( 721 (Set<E>) filtered.unfiltered, combinedPredicate); 722 } 723 724 return new FilteredSet<E>( 725 checkNotNull(unfiltered), checkNotNull(predicate)); 726 } 727 728 private static class FilteredSet<E> extends FilteredCollection<E> 729 implements Set<E> { 730 FilteredSet(Set<E> unfiltered, Predicate<? super E> predicate) { 731 super(unfiltered, predicate); 732 } 733 734 @Override public boolean equals(@Nullable Object object) { 735 return equalsImpl(this, object); 736 } 737 738 @Override public int hashCode() { 739 return hashCodeImpl(this); 740 } 741 } 742 743 /** 744 * Returns the elements of a {@code SortedSet}, {@code unfiltered}, that 745 * satisfy a predicate. The returned set is a live view of {@code unfiltered}; 746 * changes to one affect the other. 747 * 748 * <p>The resulting set's iterator does not support {@code remove()}, but all 749 * other set methods are supported. When given an element that doesn't satisfy 750 * the predicate, the set's {@code add()} and {@code addAll()} methods throw 751 * an {@link IllegalArgumentException}. When methods such as 752 * {@code removeAll()} and {@code clear()} are called on the filtered set, 753 * only elements that satisfy the filter will be removed from the underlying 754 * set. 755 * 756 * <p>The returned set isn't threadsafe or serializable, even if 757 * {@code unfiltered} is. 758 * 759 * <p>Many of the filtered set's methods, such as {@code size()}, iterate across 760 * every element in the underlying set and determine which elements satisfy 761 * the filter. When a live view is <i>not</i> needed, it may be faster to copy 762 * {@code Iterables.filter(unfiltered, predicate)} and use the copy. 763 * 764 * <p><b>Warning:</b> {@code predicate} must be <i>consistent with equals</i>, 765 * as documented at {@link Predicate#apply}. Do not provide a predicate such as 766 * {@code Predicates.instanceOf(ArrayList.class)}, which is inconsistent with 767 * equals. (See {@link Iterables#filter(Iterable, Class)} for related 768 * functionality.) 769 * 770 * @since 11.0 771 */ 772 public static <E> SortedSet<E> filter( 773 SortedSet<E> unfiltered, Predicate<? super E> predicate) { 774 return Platform.setsFilterSortedSet(unfiltered, predicate); 775 } 776 777 static <E> SortedSet<E> filterSortedIgnoreNavigable( 778 SortedSet<E> unfiltered, Predicate<? super E> predicate) { 779 if (unfiltered instanceof FilteredSet) { 780 // Support clear(), removeAll(), and retainAll() when filtering a filtered 781 // collection. 782 FilteredSet<E> filtered = (FilteredSet<E>) unfiltered; 783 Predicate<E> combinedPredicate 784 = Predicates.<E>and(filtered.predicate, predicate); 785 return new FilteredSortedSet<E>( 786 (SortedSet<E>) filtered.unfiltered, combinedPredicate); 787 } 788 789 return new FilteredSortedSet<E>( 790 checkNotNull(unfiltered), checkNotNull(predicate)); 791 } 792 793 private static class FilteredSortedSet<E> extends FilteredSet<E> 794 implements SortedSet<E> { 795 796 FilteredSortedSet(SortedSet<E> unfiltered, Predicate<? super E> predicate) { 797 super(unfiltered, predicate); 798 } 799 800 @Override 801 public Comparator<? super E> comparator() { 802 return ((SortedSet<E>) unfiltered).comparator(); 803 } 804 805 @Override 806 public SortedSet<E> subSet(E fromElement, E toElement) { 807 return new FilteredSortedSet<E>(((SortedSet<E>) unfiltered).subSet(fromElement, toElement), 808 predicate); 809 } 810 811 @Override 812 public SortedSet<E> headSet(E toElement) { 813 return new FilteredSortedSet<E>(((SortedSet<E>) unfiltered).headSet(toElement), predicate); 814 } 815 816 @Override 817 public SortedSet<E> tailSet(E fromElement) { 818 return new FilteredSortedSet<E>(((SortedSet<E>) unfiltered).tailSet(fromElement), predicate); 819 } 820 821 @Override 822 public E first() { 823 return iterator().next(); 824 } 825 826 @Override 827 public E last() { 828 SortedSet<E> sortedUnfiltered = (SortedSet<E>) unfiltered; 829 while (true) { 830 E element = sortedUnfiltered.last(); 831 if (predicate.apply(element)) { 832 return element; 833 } 834 sortedUnfiltered = sortedUnfiltered.headSet(element); 835 } 836 } 837 } 838 839 /** 840 * Returns every possible list that can be formed by choosing one element 841 * from each of the given sets in order; the "n-ary 842 * <a href="http://en.wikipedia.org/wiki/Cartesian_product">Cartesian 843 * product</a>" of the sets. For example: <pre> {@code 844 * 845 * Sets.cartesianProduct(ImmutableList.of( 846 * ImmutableSet.of(1, 2), 847 * ImmutableSet.of("A", "B", "C")))}</pre> 848 * 849 * <p>returns a set containing six lists: 850 * 851 * <ul> 852 * <li>{@code ImmutableList.of(1, "A")} 853 * <li>{@code ImmutableList.of(1, "B")} 854 * <li>{@code ImmutableList.of(1, "C")} 855 * <li>{@code ImmutableList.of(2, "A")} 856 * <li>{@code ImmutableList.of(2, "B")} 857 * <li>{@code ImmutableList.of(2, "C")} 858 * </ul> 859 * 860 * <p>The result is guaranteed to be in the "traditional", lexicographical 861 * order for Cartesian products that you would get from nesting for loops: 862 * <pre> {@code 863 * 864 * for (B b0 : sets.get(0)) { 865 * for (B b1 : sets.get(1)) { 866 * ... 867 * ImmutableList<B> tuple = ImmutableList.of(b0, b1, ...); 868 * // operate on tuple 869 * } 870 * }}</pre> 871 * 872 * <p>Note that if any input set is empty, the Cartesian product will also be 873 * empty. If no sets at all are provided (an empty list), the resulting 874 * Cartesian product has one element, an empty list (counter-intuitive, but 875 * mathematically consistent). 876 * 877 * <p><i>Performance notes:</i> while the cartesian product of sets of size 878 * {@code m, n, p} is a set of size {@code m x n x p}, its actual memory 879 * consumption is much smaller. When the cartesian set is constructed, the 880 * input sets are merely copied. Only as the resulting set is iterated are the 881 * individual lists created, and these are not retained after iteration. 882 * 883 * @param sets the sets to choose elements from, in the order that 884 * the elements chosen from those sets should appear in the resulting 885 * lists 886 * @param <B> any common base class shared by all axes (often just {@link 887 * Object}) 888 * @return the Cartesian product, as an immutable set containing immutable 889 * lists 890 * @throws NullPointerException if {@code sets}, any one of the {@code sets}, 891 * or any element of a provided set is null 892 * @since 2.0 893 */ 894 public static <B> Set<List<B>> cartesianProduct( 895 List<? extends Set<? extends B>> sets) { 896 return CartesianSet.create(sets); 897 } 898 899 /** 900 * Returns every possible list that can be formed by choosing one element 901 * from each of the given sets in order; the "n-ary 902 * <a href="http://en.wikipedia.org/wiki/Cartesian_product">Cartesian 903 * product</a>" of the sets. For example: <pre> {@code 904 * 905 * Sets.cartesianProduct( 906 * ImmutableSet.of(1, 2), 907 * ImmutableSet.of("A", "B", "C"))}</pre> 908 * 909 * <p>returns a set containing six lists: 910 * 911 * <ul> 912 * <li>{@code ImmutableList.of(1, "A")} 913 * <li>{@code ImmutableList.of(1, "B")} 914 * <li>{@code ImmutableList.of(1, "C")} 915 * <li>{@code ImmutableList.of(2, "A")} 916 * <li>{@code ImmutableList.of(2, "B")} 917 * <li>{@code ImmutableList.of(2, "C")} 918 * </ul> 919 * 920 * <p>The result is guaranteed to be in the "traditional", lexicographical 921 * order for Cartesian products that you would get from nesting for loops: 922 * <pre> {@code 923 * 924 * for (B b0 : sets.get(0)) { 925 * for (B b1 : sets.get(1)) { 926 * ... 927 * ImmutableList<B> tuple = ImmutableList.of(b0, b1, ...); 928 * // operate on tuple 929 * } 930 * }}</pre> 931 * 932 * <p>Note that if any input set is empty, the Cartesian product will also be 933 * empty. If no sets at all are provided (an empty list), the resulting 934 * Cartesian product has one element, an empty list (counter-intuitive, but 935 * mathematically consistent). 936 * 937 * <p><i>Performance notes:</i> while the cartesian product of sets of size 938 * {@code m, n, p} is a set of size {@code m x n x p}, its actual memory 939 * consumption is much smaller. When the cartesian set is constructed, the 940 * input sets are merely copied. Only as the resulting set is iterated are the 941 * individual lists created, and these are not retained after iteration. 942 * 943 * @param sets the sets to choose elements from, in the order that 944 * the elements chosen from those sets should appear in the resulting 945 * lists 946 * @param <B> any common base class shared by all axes (often just {@link 947 * Object}) 948 * @return the Cartesian product, as an immutable set containing immutable 949 * lists 950 * @throws NullPointerException if {@code sets}, any one of the {@code sets}, 951 * or any element of a provided set is null 952 * @since 2.0 953 */ 954 public static <B> Set<List<B>> cartesianProduct( 955 Set<? extends B>... sets) { 956 return cartesianProduct(Arrays.asList(sets)); 957 } 958 959 private static final class CartesianSet<E> 960 extends ForwardingCollection<List<E>> implements Set<List<E>> { 961 private transient final ImmutableList<ImmutableSet<E>> axes; 962 private transient final CartesianList<E> delegate; 963 964 static <E> Set<List<E>> create(List<? extends Set<? extends E>> sets) { 965 ImmutableList.Builder<ImmutableSet<E>> axesBuilder = 966 new ImmutableList.Builder<ImmutableSet<E>>(sets.size()); 967 for (Set<? extends E> set : sets) { 968 ImmutableSet<E> copy = ImmutableSet.copyOf(set); 969 if (copy.isEmpty()) { 970 return ImmutableSet.of(); 971 } 972 axesBuilder.add(copy); 973 } 974 final ImmutableList<ImmutableSet<E>> axes = axesBuilder.build(); 975 ImmutableList<List<E>> listAxes = new ImmutableList<List<E>>() { 976 977 @Override 978 public int size() { 979 return axes.size(); 980 } 981 982 @Override 983 public List<E> get(int index) { 984 return axes.get(index).asList(); 985 } 986 987 @Override 988 boolean isPartialView() { 989 return true; 990 } 991 }; 992 return new CartesianSet<E>(axes, new CartesianList<E>(listAxes)); 993 } 994 995 private CartesianSet( 996 ImmutableList<ImmutableSet<E>> axes, CartesianList<E> delegate) { 997 this.axes = axes; 998 this.delegate = delegate; 999 } 1000 1001 @Override 1002 protected Collection<List<E>> delegate() { 1003 return delegate; 1004 } 1005 1006 @Override public boolean equals(@Nullable Object object) { 1007 // Warning: this is broken if size() == 0, so it is critical that we 1008 // substitute an empty ImmutableSet to the user in place of this 1009 if (object instanceof CartesianSet) { 1010 CartesianSet<?> that = (CartesianSet<?>) object; 1011 return this.axes.equals(that.axes); 1012 } 1013 return super.equals(object); 1014 } 1015 1016 @Override 1017 public int hashCode() { 1018 // Warning: this is broken if size() == 0, so it is critical that we 1019 // substitute an empty ImmutableSet to the user in place of this 1020 1021 // It's a weird formula, but tests prove it works. 1022 int adjust = size() - 1; 1023 for (int i = 0; i < axes.size(); i++) { 1024 adjust *= 31; 1025 adjust = ~~adjust; 1026 // in GWT, we have to deal with integer overflow carefully 1027 } 1028 int hash = 1; 1029 for (Set<E> axis : axes) { 1030 hash = 31 * hash + (size() / axis.size() * axis.hashCode()); 1031 1032 hash = ~~hash; 1033 } 1034 hash += adjust; 1035 return ~~hash; 1036 } 1037 } 1038 1039 /** 1040 * Returns the set of all possible subsets of {@code set}. For example, 1041 * {@code powerSet(ImmutableSet.of(1, 2))} returns the set {@code {{}, 1042 * {1}, {2}, {1, 2}}}. 1043 * 1044 * <p>Elements appear in these subsets in the same iteration order as they 1045 * appeared in the input set. The order in which these subsets appear in the 1046 * outer set is undefined. Note that the power set of the empty set is not the 1047 * empty set, but a one-element set containing the empty set. 1048 * 1049 * <p>The returned set and its constituent sets use {@code equals} to decide 1050 * whether two elements are identical, even if the input set uses a different 1051 * concept of equivalence. 1052 * 1053 * <p><i>Performance notes:</i> while the power set of a set with size {@code 1054 * n} is of size {@code 2^n}, its memory usage is only {@code O(n)}. When the 1055 * power set is constructed, the input set is merely copied. Only as the 1056 * power set is iterated are the individual subsets created, and these subsets 1057 * themselves occupy only a small constant amount of memory. 1058 * 1059 * @param set the set of elements to construct a power set from 1060 * @return the power set, as an immutable set of immutable sets 1061 * @throws IllegalArgumentException if {@code set} has more than 30 unique 1062 * elements (causing the power set size to exceed the {@code int} range) 1063 * @throws NullPointerException if {@code set} is or contains {@code null} 1064 * @see <a href="http://en.wikipedia.org/wiki/Power_set">Power set article at 1065 * Wikipedia</a> 1066 * @since 4.0 1067 */ 1068 @GwtCompatible(serializable = false) 1069 public static <E> Set<Set<E>> powerSet(Set<E> set) { 1070 return new PowerSet<E>(set); 1071 } 1072 1073 private static final class SubSet<E> extends AbstractSet<E> { 1074 private final ImmutableMap<E, Integer> inputSet; 1075 private final int mask; 1076 1077 SubSet(ImmutableMap<E, Integer> inputSet, int mask) { 1078 this.inputSet = inputSet; 1079 this.mask = mask; 1080 } 1081 1082 @Override 1083 public Iterator<E> iterator() { 1084 return new UnmodifiableIterator<E>() { 1085 final ImmutableList<E> elements = inputSet.keySet().asList(); 1086 int remainingSetBits = mask; 1087 1088 @Override 1089 public boolean hasNext() { 1090 return remainingSetBits != 0; 1091 } 1092 1093 @Override 1094 public E next() { 1095 int index = Integer.numberOfTrailingZeros(remainingSetBits); 1096 if (index == 32) { 1097 throw new NoSuchElementException(); 1098 } 1099 remainingSetBits &= ~(1 << index); 1100 return elements.get(index); 1101 } 1102 }; 1103 } 1104 1105 @Override 1106 public int size() { 1107 return Integer.bitCount(mask); 1108 } 1109 1110 @Override 1111 public boolean contains(@Nullable Object o) { 1112 Integer index = inputSet.get(o); 1113 return index != null && (mask & (1 << index)) != 0; 1114 } 1115 } 1116 1117 private static final class PowerSet<E> extends AbstractSet<Set<E>> { 1118 final ImmutableMap<E, Integer> inputSet; 1119 1120 PowerSet(Set<E> input) { 1121 ImmutableMap.Builder<E, Integer> builder = ImmutableMap.builder(); 1122 int i = 0; 1123 for (E e : checkNotNull(input)) { 1124 builder.put(e, i++); 1125 } 1126 this.inputSet = builder.build(); 1127 checkArgument(inputSet.size() <= 30, 1128 "Too many elements to create power set: %s > 30", inputSet.size()); 1129 } 1130 1131 @Override public int size() { 1132 return 1 << inputSet.size(); 1133 } 1134 1135 @Override public boolean isEmpty() { 1136 return false; 1137 } 1138 1139 @Override public Iterator<Set<E>> iterator() { 1140 return new AbstractIndexedListIterator<Set<E>>(size()) { 1141 @Override protected Set<E> get(final int setBits) { 1142 return new SubSet<E>(inputSet, setBits); 1143 } 1144 }; 1145 } 1146 1147 @Override public boolean contains(@Nullable Object obj) { 1148 if (obj instanceof Set) { 1149 Set<?> set = (Set<?>) obj; 1150 return inputSet.keySet().containsAll(set); 1151 } 1152 return false; 1153 } 1154 1155 @Override public boolean equals(@Nullable Object obj) { 1156 if (obj instanceof PowerSet) { 1157 PowerSet<?> that = (PowerSet<?>) obj; 1158 return inputSet.equals(that.inputSet); 1159 } 1160 return super.equals(obj); 1161 } 1162 1163 @Override public int hashCode() { 1164 /* 1165 * The sum of the sums of the hash codes in each subset is just the sum of 1166 * each input element's hash code times the number of sets that element 1167 * appears in. Each element appears in exactly half of the 2^n sets, so: 1168 */ 1169 return inputSet.keySet().hashCode() << (inputSet.size() - 1); 1170 } 1171 1172 @Override public String toString() { 1173 return "powerSet(" + inputSet + ")"; 1174 } 1175 } 1176 1177 /** 1178 * An implementation for {@link Set#hashCode()}. 1179 */ 1180 static int hashCodeImpl(Set<?> s) { 1181 int hashCode = 0; 1182 for (Object o : s) { 1183 hashCode += o != null ? o.hashCode() : 0; 1184 1185 hashCode = ~~hashCode; 1186 // Needed to deal with unusual integer overflow in GWT. 1187 } 1188 return hashCode; 1189 } 1190 1191 /** 1192 * An implementation for {@link Set#equals(Object)}. 1193 */ 1194 static boolean equalsImpl(Set<?> s, @Nullable Object object) { 1195 if (s == object) { 1196 return true; 1197 } 1198 if (object instanceof Set) { 1199 Set<?> o = (Set<?>) object; 1200 1201 try { 1202 return s.size() == o.size() && s.containsAll(o); 1203 } catch (NullPointerException ignored) { 1204 return false; 1205 } catch (ClassCastException ignored) { 1206 return false; 1207 } 1208 } 1209 return false; 1210 } 1211 1212 /** 1213 * Remove each element in an iterable from a set. 1214 */ 1215 static boolean removeAllImpl(Set<?> set, Iterator<?> iterator) { 1216 boolean changed = false; 1217 while (iterator.hasNext()) { 1218 changed |= set.remove(iterator.next()); 1219 } 1220 return changed; 1221 } 1222 1223 static boolean removeAllImpl(Set<?> set, Collection<?> collection) { 1224 checkNotNull(collection); // for GWT 1225 if (collection instanceof Multiset) { 1226 collection = ((Multiset<?>) collection).elementSet(); 1227 } 1228 /* 1229 * AbstractSet.removeAll(List) has quadratic behavior if the list size 1230 * is just less than the set's size. We augment the test by 1231 * assuming that sets have fast contains() performance, and other 1232 * collections don't. See 1233 * http://code.google.com/p/guava-libraries/issues/detail?id=1013 1234 */ 1235 if (collection instanceof Set && collection.size() > set.size()) { 1236 return Iterators.removeAll(set.iterator(), collection); 1237 } else { 1238 return removeAllImpl(set, collection.iterator()); 1239 } 1240 } 1241 }